package cn.miaoyao.miaoyaoms.utils;
import cn.miaoyao.miaoyaoms.Handler.ExcelListener;
import cn.miaoyao.miaoyaoms.service.MiaosampleService;
import com.alibaba.excel.EasyExcel;
import io.netty.util.internal.StringUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.util.Arrays;
import java.util.List;

/**
 * @Author: shushu
 * @Date: 2023/05/17/20:45
 * @Description:Excel的工具类
 */
@Component
@Slf4j
public class ExcelUtil<T> {
    //excel文件后缀
    private final static String xls="xls";
    private final static String xlsx="xlsx";

    //校验文件后缀是否为xls，xlsx
    public static boolean checkExcelExtension(MultipartFile excel){
        String filename = excel.getOriginalFilename();//获取上传文件的原来名称
        if(StringUtil.isNullOrEmpty(filename)){
            log.info("excel文件名为null");
            return false;
        }
        int index = filename.lastIndexOf(".");
        if(index==-1){
            log.info("excel文件名中没有点号");
            return false;
        }
        String extension = filename.substring(index + 1);
        return Arrays.asList(xls,xlsx).contains(extension);
    }

    /**
     * 读取excel文件
     * @param inputStream 文件输入流
     * @param clazz excel文件对应的实体类。主要是将读取的文件数据封装到实体类中。
     * @return
     */
    public Boolean simpleExcelRead(InputStream inputStream, Class<T> clazz, MiaosampleService miaosampleService){
        ExcelListener excelListener = new ExcelListener(miaosampleService);
        EasyExcel.read(inputStream,clazz,excelListener).sheet().doRead();//读取excel文件，然后将数据封装到实体类，然后调用监听器中的invoke方法处理数据，同时将数据存入dataList中。
        if(excelListener.getDataList().size()==0)
            return true;
        else
            return false;
    }

}
